package com.c2b.algorithm.leetcode.jzoffer;

import java.util.Stack;

/**
 * <a href="https://www.nowcoder.com/practice/54275ddae22f475981afa2244dd448c6?tpId=13&&tqId=11158&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking">用两个栈实现队列</a>
 * <p>
 * 用两个栈来实现一个队列，使用n个元素来完成 n 次在队列尾部插入整数(push)和n次在队列头部删除整数(pop)的功能。 队列中的元素为int类型。
 * 保证操作合法，即保证pop操作时队列内已有元素。
 * </p>
 *
 * @author c2b
 * @since 2023/3/7 18:07
 */
public class JzOffer0009QueueThroughTwoStacks {
    Stack<Integer> stack1 = new Stack<>();
    Stack<Integer> stack2 = new Stack<>();

    /**
     * 当插入时，直接插入 stack1
     *
     * @param node 压入栈的元素
     */
    public void push(int node) {
        stack1.push(node);
    }

    /**
     * 当弹出时，当 stack2 不为空，弹出 stack2 栈顶元素，如果 stack2 为空，将 stack1 中的全部数逐个出栈入栈 stack2，再弹出 stack2 栈顶元素
     *
     * @return 弹出栈的元素
     */
    public int pop() {
        if (!stack2.empty()) {
            return stack2.pop();
        }
        while (!stack1.empty()) {
            stack2.push(stack1.pop());
        }
        return stack2.pop();
    }

    public static void main(String[] args) {
        JzOffer0009QueueThroughTwoStacks jzOffer0009QueueThroughTwoStacks = new JzOffer0009QueueThroughTwoStacks();
        jzOffer0009QueueThroughTwoStacks.push(2);
        jzOffer0009QueueThroughTwoStacks.push(3);
        System.out.println(jzOffer0009QueueThroughTwoStacks.pop());
        jzOffer0009QueueThroughTwoStacks.push(1);
        System.out.println(jzOffer0009QueueThroughTwoStacks.pop());
        System.out.println(jzOffer0009QueueThroughTwoStacks.pop());
    }
}
